# -*- coding: utf-8 -*-
"""
    @Author：SimpleTree
    @date：2025/6/5 11:21
    @desc:
"""
from django.db import models


# VectorField 是一个自定义 Django 字段，用于支持 PostgreSQL 的 vector 数据类型。
# 这个字段允许你在 Django 模型中直接使用向量存储功能，非常适合构建基于向量相似度的应用（如语义搜索、推荐系统等）。
class VectorField(models.Field):
    # 重写 db_type 方法，告诉 Django 在数据库中创建 vector 类型的字段。这是集成 PostgreSQL pgvector 扩展的关键步骤。

    def db_type(self, connection):
        return "vector"


class TsVectorField(models.Field):
    def db_type(self, connection):
        return "tsvector"
